package com.base.dynamicprograming;

/**
 * 714. 买卖股票的最佳时机含手续费
 * 给定一个整数数组 prices，其中第 i 个元素代表了第 i 天的股票价格 ；整数 fee 代表了交易股票的手续费用。
 * <p>
 * 你可以无限次地完成交易，但是你每笔交易都需要付手续费。如果你已经购买了一个股票，在卖出它之前你就不能再继续购买股票了。
 * <p>
 * 返回获得利润的最大值。
 * <p>
 * 注意：这里的一笔交易指买入持有并卖出股票的整个过程，每笔交易你只需要为支付一次手续费
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/best-time-to-buy-and-sell-stock-with-transaction-fee
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 *
 * @author leon
 * @date 2021年12月04日 14:42
 */
public class MaxProfit4 {
    /**
     * 状态一，未持有状态
     * 状态二，持有状态
     *
     * @param prices
     * @param fee
     * @return
     */
    public int maxProfit(int[] prices, int fee) {
        int n = prices.length;
        int[] states = new int[2];
        states[1] = -prices[0];

        for (int i = 1; i < n; i++) {
            // 未持有状态
            int pre0 = states[0];
            // 持有状态
            int pre1 = states[1];

            states[0] = Math.max(pre0, pre1 + prices[i] - fee);
            states[1] = Math.max(pre1, pre0 - prices[i]);
        }

        return Math.max(states[0], states[1]);
    }
}
